#!/bin/bash

# 安装 OpenGL 驱动和工具
echo "安装 OpenGL 驱动和测试工具..."
sudo apt-get update
sudo apt-get install -y libgl1-mesa-glx mesa-utils

# 检查 GPU 是否启用成功
echo "检查 OpenGL 和 GPU 驱动..."
glxinfo_output=$(glxinfo | grep "OpenGL" 2>/dev/null)

# 打印 glxinfo | grep "OpenGL" 的原始输出
echo "原始 OpenGL 信息输出："
echo "$glxinfo_output"

# 判断是否包含 NVIDIA 驱动信息
if echo "$glxinfo_output" | grep -q "NVIDIA Corporation"; then
    echo "GPU 启用成功！检测到 NVIDIA 驱动。"
else
    echo "GPU 启用失败，未检测到 NVIDIA 驱动。请检查 NVIDIA 驱动和 Docker 配置。"
fi

# 将以下参数插入你正常使用的命令中，即可启用GPU（--privileged可省略）
# --runtime=nvidia --gpus all -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all -e LIBGL_ALWAYS_INDIRECT=1 -e DRI_PRIME=1 -e __GLX_VENDOR_LIBRARY_NAME=nvidia -e USER=$(whoami) -v /tmp/.X11-unix:/tmp/.X11-unix --privileged
